package com.xxxx.server.config;

import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.xxxx.server.pojo.MailConstants;
import com.xxxx.server.pojo.MailLog;
import com.xxxx.server.service.IMailLogService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
@Slf4j
public class RabbitMqCallBack implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {
    @Autowired
    private RabbitTemplate rabbitTemplate;
    @Autowired
    private IMailLogService mailLogService;

    @PostConstruct
    public void init() {
        // 注入 手动发布确认
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 1.发消息接收到了 回调
     *   1.1 correlationData 保存回调消息的ID及相关消息
     *   1.2 交换机收到信息 ack=true
     *   1.3 cause null
     * 2.接收不到 回调
     *   2.1 correlationData 保存回调消息的ID及相关信息
     *   2.2 交换机收不到信息 ack=false
     * */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String msgId = correlationData.getId();
        if (ack == true) {
            // 根据msgId去更新消息投递状态为成功
            mailLogService.update(new UpdateWrapper<MailLog>().set("status", MailConstants.SUCCESS)
                    .eq("msgId", msgId));
            log.info("{}====>消息发送成功",msgId);
        } else {
            log.info("{}====>消息发送失败",msgId);
        }
    }

    // 可以在当消息传递过程中不可达目的地时将消息返回给生产者
    // 只有不可达目的地时才进行回退
    // (经过测试有小问题，消息被交换机退回时，status仍旧因为消息发出去了而被改成了成功，在这也拿不到消息的msgid，才疏学浅，无法解决 ┭┮﹏┭┮)
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.info("消息：{}，被交换机{}退回，退回原因：{}，路由key：{}",
                new String(message.getBody()),exchange,replyText,routingKey);
    }
}